Skip to content

Remove input_tensor_desc destruction in RNN kernels#729

Open
magaonka-amd wants to merge 1 commit intoROCm:rocm-jaxlib-v0.9.0from
magaonka-amd:fix/rnn-hip-memleak-v0.9.0
Open

Remove input_tensor_desc destruction in RNN kernels#729
magaonka-amd wants to merge 1 commit intoROCm:rocm-jaxlib-v0.9.0from
magaonka-amd:fix/rnn-hip-memleak-v0.9.0

Conversation

@magaonka-amd
Copy link

The input_tensor_desc is still needed by MIOpen for the execution buffer and must not be destroyed before the RNN operation completes.

Thanks to Kamil for pointing that out.

The input_tensor_desc is still needed by MIOpen for the execution
buffer and must not be destroyed before the RNN operation completes.
@magaonka-amd magaonka-amd added the cherry-pick-candidate Mark a PR to be cherry-picked into the next ROCm JAX. Remove IIF the latest upstream contain the PR. label Mar 5, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

cherry-pick-candidate Mark a PR to be cherry-picked into the next ROCm JAX. Remove IIF the latest upstream contain the PR.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant